# 简单的登录功能

# 模拟的用户数据（通常可以从数据库或文件中加载）
users = {
    "admin": "123456",  # 用户名: 密码
    "user1": "mypassword",
    "user2": "securepass"
}

def login():
    print("=== 登录系统 ===")
    username = input("请输入用户名: ").strip()
    password = input("请输入密码: ").strip()

    # 验证用户名和密码
    if username in users and users[username] == password:
        print("登录成功！欢迎回来，{}！".format(username))
    else:
        print("登录失败！用户名或密码错误。")
def add(a,b):
    return a+b

def sub(a,b):
    return a-b

def main():
    while True:
        print("\n请选择操作：")
        print("1. 登录")
        print("2. 退出")
        choice = input("请输入选项 (1/2): ").strip()

        if choice == "1":
            login()
        elif choice == "2":
            print("退出程序。再见！")
            break
        else:
            print("无效选项，请重新选择。")

if __name__ == "__main__":
    main()